from modelscope.msdatasets import MsDataset
ds_train =  MsDataset.load('modelscope/imdb', subset_name='plain_text', split='train')
ds_test =  MsDataset.load('modelscope/imdb', subset_name='plain_text', split='model_test')